# 封装一个断言的工具类
class AssertUtil:

    # 封装一个断言客达天下的方法
    @classmethod
    def assert_kdtx(cls, resp, status_code, msg, code):
        assert status_code == resp.status_code
        assert msg == resp.json().get("msg")
        assert code == resp.json().get("code")

    @classmethod
    def assert_ihrm(cls,resp,status_code,success,code,message):
        assert status_code == resp.status_code
        assert success == resp.json().get("success")
        assert code == resp.json().get("code")
        assert message == resp.json().get("message")
